-
Notifications
You must be signed in to change notification settings - Fork 516
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
pt: refact training code #3359
pt: refact training code #3359
Conversation
Signed-off-by: Duo <[email protected]>
for more information, see https://pre-commit.ci
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## devel #3359 +/- ##
==========================================
+ Coverage 76.06% 76.53% +0.47%
==========================================
Files 421 421
Lines 35514 35638 +124
Branches 1618 1618
==========================================
+ Hits 27013 27276 +263
+ Misses 7633 7494 -139
Partials 868 868 ☔ View full report in Codecov by Sentry. |
@@ -83,7 +83,7 @@ | |||
param.requires_grad = trainable | |||
|
|||
def share_params(self, shared_links, resume=False): | |||
supported_types = ["type_embedding", "descriptor", "fitting_net"] | |||
supported_types = ["descriptor", "fitting_net"] |
Check notice
Code scanning / CodeQL
Unused local variable Note
Signed-off-by: Han Wang <[email protected]>
for more information, see https://pre-commit.ci
Signed-off-by: Duo <[email protected]>
for more information, see https://pre-commit.ci
This PR
make_stat_input
and related training code